package algorithm.middle;

import java.awt.image.DirectColorModel;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;

/**
 * 机器人在一个无限大小的 XY 网格平面上行走，从点 (0, 0) 处开始出发，面向北方。该机器人可以接收以下三种类型的命令 commands ：
 * <p>
 * -2 ：向左转 90 度
 * -1 ：向右转 90 度
 * 1 <= x <= 9 ：向前移动 x 个单位长度
 * 在网格上有一些格子被视为障碍物 obstacles 。第 i 个障碍物位于网格点  obstacles[i] = (xi, yi) 。
 * <p>
 * 机器人无法走到障碍物上，它将会停留在障碍物的前一个网格方块上，但仍然可以继续尝试进行该路线的其余部分。
 * <p>
 * 返回从原点到机器人所有经过的路径点（坐标为整数）的最大欧式距离的平方。（即，如果距离为 5 ，则返回 25 ）
 */
public class RobotSim874 {
    public HashMap<Integer, Set<Integer>> map;

    public int robotSim(int[] commands, int[][] obstacles) {
        map = new HashMap<>();
        for (int[] obstacle : obstacles) {
            map.computeIfAbsent(obstacle[0], key -> new HashSet<>()).add(obstacle[1]);
        }
        //东1 南2 西3 北4
        int direction = 4;
        int res = 0;
        int x = 0;
        int y = 0;
        for (int command : commands) {
            if (command == -1) {
                //右转
                direction = (direction == 4) ? 1 : (direction + 1);
            } else if (command == -2) {
                //左转
                direction = (direction == 1) ? 4 : (direction - 1);
            } else {
                //直行
                if (direction == 1) {
                    //东 x++
                    for (int i = 0; i < command; i++) {
                        x++;
                        if (isObstacles(x, y)) {
                            x--;
                            break;
                        } else {
                            res = Math.max(res, x * x + y * y);
                        }
                    }
                } else if (direction == 2) {
                    //南 y--
                    for (int i = 0; i < command; i++) {
                        y--;
                        if (isObstacles(x, y)) {
                            y++;
                            break;
                        } else {
                            res = Math.max(res, x * x + y * y);
                        }
                    }
                } else if (direction == 3) {
                    //西 x--
                    for (int i = 0; i < command; i++) {
                        x--;
                        if (isObstacles(x, y)) {
                            x++;
                            break;
                        } else {
                            res = Math.max(res, x * x + y * y);
                        }
                    }
                } else {
                    //北 y++
                    for (int i = 0; i < command; i++) {
                        y++;
                        if (isObstacles(x, y)) {
                            y--;
                            break;
                        } else {
                            res = Math.max(res, x * x + y * y);
                        }
                    }
                }

            }
        }
        return res;

    }

    public boolean isObstacles(int x, int y) {
        Set<Integer> integers = map.get(x);
        if (integers != null) {
            return integers.contains(y);
        }
        return false;
    }

    public static void main(String[] args) {
        new RobotSim874().robotSim(new int[]{6, -1, -1, 6}, new int[][]{});
    }


}
